package cn.element.io.nio.channel;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.Iterator;

/**
 * 实现NIO非阻塞通信模式
 */
public class Server {

    public static void main(String[] args) throws IOException {
        // 1.获取通道
        ServerSocketChannel ssc = ServerSocketChannel.open();
        
        // 2.切换为非阻塞模式
        ssc.configureBlocking(false);
        
        // 3.绑定连接的端口
        ssc.bind(new InetSocketAddress(8848));
        
        // 4.获取选择器Selector
        Selector selector = Selector.open();
        
        // 5.将通道都注册到选择器上.并且开始指定监听事件
        ssc.register(selector, SelectionKey.OP_ACCEPT);
        
        // 6.使用Selector轮询已经就绪好的事件

        while (selector.select() > 0) {
            System.out.println("有事件就绪了。。。。。");
            // 7.获取选择器中所有已经就绪好的事件
            Iterator<SelectionKey> it = selector.selectedKeys().iterator();

            // 8.开始遍历这些准备好的事件
            while (it.hasNext()) {
                SelectionKey key = it.next();

                // 9.判断这个事件是什么
                if (key.isAcceptable()) {
                    System.out.println("accept 事件。。。。");
                    // 10.直接获取当前接入的客户端通道
                    SocketChannel skc = ssc.accept();

                    // 11.切换成非阻塞模式
                    skc.configureBlocking(false);

                    // 12.将本客户端通道注册到选择器
                    skc.register(selector, SelectionKey.OP_READ);
                } else if (key.isReadable()) {
                    System.out.println("读事件。。。。。");
                    // 13.获取当前选择器上的读就绪事件
                    SocketChannel skc = (SocketChannel) key.channel();
                    
                    // 14.读取数据
                    ByteBuffer buffer = ByteBuffer.allocate(1024);
                    
                    int len;
                    while ((len = skc.read(buffer)) > 0) {
                        buffer.flip();
                        System.out.println(new String(buffer.array(), 0, len));
                        buffer.clear();  // 清除之前的数据
                    }
                }
                
                it.remove();
            }
            
        }
    }

}
